import argparse
import torch

parser = argparse.ArgumentParser(description="LiveDeep with PyTorch")
parser.add_argument('--cuda', default=torch.cuda.is_available(), type=bool, help='whether cuda is in use')
parser.add_argument('--tile_x', default=5, type=int, help='tile x num in width')
parser.add_argument('--tile_y', default=5, type=int, help='tile y num in height')
parser.add_argument('--view_x', default=2, type=int, help='tile y num in height')
parser.add_argument('--view_y', default=2, type=int, help='tile y num in height')


parser.add_argument('--lr', default=0.0001, type=float, help='learning rate')
parser.add_argument('--lrLSTM', default=0.01, type=float, help='learning rate for LSTM')
parser.add_argument('--thresLoss', default=0.00001, type=float, help='threshold loss for LSTM')
parser.add_argument('--LSTMflag', default=False, type=bool, help='whether to use LSTM')
parser.add_argument('--thresFactor', default=0, type=int, help='the scale of the threshold')
parser.add_argument('--epochCNN', default=10, type=int, help='the Epochs for CNN')


parser.add_argument('--log_path', default="D:/VR_project/LiveDeep_All/LiveDeep/normal/5x5/",
                    type=str, help='save log path')


def get_args():
    args = parser.parse_args()
    return args